package cn.bugstack.chatgpt.data.types.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;

@Getter
@AllArgsConstructor
public enum DeepSeekModel {
    DEEPSEEK_V3("deepseek-chat"),
    DEEPSEEK_R1("deepseek-reasoner"),

    ;
    private String code;

    public static DeepSeekModel get(String code){
        switch(code){
            case "deepseek-reasoner":
                return DeepSeekModel.DEEPSEEK_R1;
            case "deepseek-chat":
                return DeepSeekModel.DEEPSEEK_V3;
            default:
                return DeepSeekModel.DEEPSEEK_V3;
        }
    }


}
